<?php

// Copyright 2020 Catchpoint Systems Inc.
// Use of this source code is governed by the Polyform Shield 1.0.0 license that can be
// found in the LICENSE.md file.

require_once INCLUDES_PATH . '/include/TestPaths.php';
require_once INCLUDES_PATH . '/include/UrlGenerator.php';

// TODO: get rid of this in the long run (for xml_entities)
require_once INCLUDES_PATH . '/common_lib.inc';

class XmlResultGenerator
{
    const INFO_REQUESTS = 1;
    const INFO_MEDIAN_REQUESTS = 2;
    const INFO_DOMAIN_BREAKDOWN = 3;
    const INFO_MIMETYPE_BREAKDOWN = 4;
    const INFO_CONSOLE = 5;

  /**
   * @var TestInfo Information about the test
   */
    private $testInfo;
    private $baseUrl;
    private $additionalInfo;
    private $fileHandler;
    private $friendlyUrls;
    private $forceMultistep;

  /**
   * XmlResultGenerator constructor.
   * @param TestInfo $testInfo Information about the test
   * @param string $urlStart Start for test-related URLs
   * @param FileHandler $fileHandler FileHandler to be used
   * @param array $additionalInfo Array of INFO_* constants to define which additional information should be printed
   * @param bool $friendlyUrls True if friendly URLs should be used (mod_rewrite), false otherwise
   */
    public function __construct($testInfo, $urlStart, $fileHandler, $additionalInfo, $friendlyUrls)
    {
        $this->testInfo = $testInfo;
        $this->baseUrl = $urlStart;
        $this->additionalInfo = $additionalInfo;
        $this->fileHandler = $fileHandler;
        $this->friendlyUrls = $friendlyUrls;
        $this->forceMultistep = false;
    }

  /**
   * For singlestep measurement, the output is still in singlestep format. With this method, the behavior can
   * be changed.
   * @param $force True if the output should be always in multistep format, false otherwise.
   */
    public function forceMultistepFormat($force)
    {
        $this->forceMultistep = $force;
    }

  /**
   * @param TestResults $testResults
   * @param string $median_metric
   * @param string $requestId
   * @param bool $medianFvOnly
   */
    public function printAllResults($testResults, $median_metric, $requestId = null, $medianFvOnly = false)
    {
        $urlGenerator = UrlGenerator::create($this->friendlyUrls, $this->baseUrl, $this->testInfo->getId(), 0, 0);

        echo "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n";
        echo "<response>\n";
        echo "<statusCode>200</statusCode>\n";
        echo "<statusText>Ok</statusText>\n";
        if (!empty($requestId)) {
            echo "<requestId>$requestId</requestId>\n";
        }
        if (defined("VER_WEBPAGETEST")) {
            echo "<webPagetestVersion>" . VER_WEBPAGETEST . "</webPagetestVersion>";
        }
        echo "<data>\n";

        echo "<testId>" . $this->testInfo->getId() . "</testId>\n";
        echo "<summary>" . $urlGenerator->resultSummary() . "</summary>\n";

        $testInfo = $this->testInfo->getInfoArray();
        if ($testInfo) {
            if (@strlen($testInfo['url'])) {
                echo "<testUrl>" . xml_entities($testInfo['url']) . "</testUrl>\n";
            }
            if (@strlen($testInfo['location'])) {
                $locstring = $testInfo['location'];
                if (@strlen($testInfo['browser'])) {
                    $locstring .= ':' . $testInfo['browser'];
                }
                echo "<location>$locstring</location>\n";
            }
            $location = $this->testInfo->getTestLocation();
            if ($location) {
                echo "<from>" . xml_entities($location) . "</from>\n";
            }
            if (@strlen($testInfo['connectivity'])) {
                echo "<connectivity>{$testInfo['connectivity']}</connectivity>\n";
                echo "<bwDown>{$testInfo['bwIn']}</bwDown>\n";
                echo "<bwUp>{$testInfo['bwOut']}</bwUp>\n";
                echo "<latency>{$testInfo['latency']}</latency>\n";
                echo "<plr>{$testInfo['plr']}</plr>\n";
            }
            if (isset($testInfo['mobile'])) {
                echo "<mobile>" . xml_entities($testInfo['mobile']) .   "</mobile>\n";
            }
            if (@strlen($testInfo['label'])) {
                echo "<label>" . xml_entities($testInfo['label']) . "</label>\n";
            }
            if (@strlen($testInfo['completed'])) {
                echo "<completed>" . gmdate("r", $testInfo['completed']) . "</completed>\n";
            }
            if (@strlen($testInfo['tester'])) {
                echo "<tester>" . xml_entities($testInfo['tester']) . "</tester>\n";
            }
            if (@strlen($testInfo['testerDNS'])) {
                echo "<testerDNS>" . xml_entities($testInfo['testerDNS']) . "</testerDNS>\n";
            }
        }

      // spit out the calculated averages
        $fv = $testResults->getFirstViewAverage();
        $rv = $testResults->getRepeatViewAverage();
        $runs = $testResults->countRuns();

        echo "<runs>$runs</runs>\n";
        echo "<successfulFVRuns>" . $testResults->countSuccessfulRuns(false) . "</successfulFVRuns>\n";
        if (isset($rv)) {
            echo "<successfulRVRuns>" . $testResults->countSuccessfulRuns(true) . "</successfulRVRuns>\n";
        }

        echo "<average>\n";
        echo "<firstView>\n";
        foreach ($fv as $key => $val) {
            $key = preg_replace('/[^a-zA-Z0-9\.\-_]/', '_', $key);
            echo "<$key>" . number_format($val, 0, '.', '') . "</$key>\n";
        }
        echo "</firstView>\n";
        if (isset($rv)) {
            echo "<repeatView>\n";
            foreach ($rv as $key => $val) {
                $key = preg_replace('/[^a-zA-Z0-9\.\-_]/', '_', $key);
                echo "<$key>" . number_format($val, 0, '.', '') . "</$key>\n";
            }
            echo "</repeatView>\n";
        }
        echo "</average>\n";
        echo "<standardDeviation>\n";
        echo "<firstView>\n";
        foreach ($fv as $key => $val) {
            $key = preg_replace('/[^a-zA-Z0-9\.\-_]/', '_', $key);
            echo "<$key>" . $testResults->getStandardDeviation($key, false) . "</$key>\n";
        }
        echo "</firstView>\n";
        if (isset($rv)) {
            echo "<repeatView>\n";
            foreach ($rv as $key => $val) {
                $key = preg_replace('/[^a-zA-Z0-9\.\-_]/', '_', $key);
                echo "<$key>" . $testResults->getStandardDeviation($key, true) . "</$key>\n";
            }
            echo "</repeatView>\n";
        }
        echo "</standardDeviation>\n";

      // output the median run data
        $medianMode = (!empty($_REQUEST["medianRun"]) && $_REQUEST["medianRun"] == "fastest") ? "fastest" : "median";
        $fvMedian = $testResults->getMedianRunNumber($median_metric, false, $medianMode);
        if ($fvMedian) {
            echo "<median>\n";
            $this->printMedianRun($testResults->getRunResult($fvMedian, false));

            if (isset($rv)) {
                $rvMedian = $medianFvOnly ? $fvMedian : $testResults->getMedianRunNumber($median_metric, true, $medianMode);
                if ($rvMedian) {
                    $this->printMedianRun($testResults->getRunResult($rvMedian, true));
                }
            }
            echo "</median>\n";
        }

      // spit out the raw data for each run
        for ($i = 1; $i <= $runs; $i++) {
            echo "<run>\n";
            echo "<id>$i</id>\n";

            $this->printRun($testResults->getRunResult($i, false));
            $this->printRun($testResults->getRunResult($i, true));

            echo "</run>\n";
        }

        echo "</data>\n";
        echo "</response>\n";
    }

  /**
   * @param TestRunResults $testResult Result for the median run
   */
    public function printMedianRun($testResult)
    {
        $run = $testResult->getRunNumber();

        $this->printViewRootStartTag($testResult->isCachedRun());
        echo "<run>" . $run . "</run>\n";
        $this->printTester($run);

        if ($this->forceMultistep || $testResult->countSteps() > 1) {
            echo ArrayToXML($testResult->aggregateRawResults());
        } else {
            $singlestepResult = $testResult->getStepResult(1);
            echo ArrayToXML($singlestepResult->getRawResults());
            $this->printAdditionalInformation($singlestepResult, true);
        }

        $this->printViewRootEndTag($testResult->isCachedRun());
    }

  /**
   * @param TestRunResults $runResult Result of this run
   */
    public function printRun($runResult)
    {
        if (empty($runResult)) {
            return;
        }
        $testResult = $runResult->getStepResult(1);
        $numSteps = $runResult->countSteps();

        $this->printViewRootStartTag($testResult->isCachedRun());
        $this->printTester($runResult->getRunNumber());
        echo "<numSteps>" . $numSteps . "</numSteps>\n";

        if ($this->forceMultistep || $numSteps > 1) {
            for ($step = 1; $step <= $numSteps; $step++) {
                $testStepResult = $runResult->getStepResult($step);
                $eventName = empty($testStepResult) ? "" : $testStepResult->getEventName();
                echo "<step>\n";
                echo "<id>" . $step . "</id>";
                echo "<eventName>" . $eventName . "</eventName>";
                $this->printStepResults($testStepResult);
                echo "</step>\n";
            }
        } else {
            $this->printStepResults($runResult->getStepResult(1));
        }

        $this->printViewRootEndTag($testResult->isCachedRun());
    }

  /**
   * @param TestStepResult $stepResult Results for the step to be printed
   */
    private function printStepResults($stepResult)
    {
        if (empty($stepResult)) {
            return;
        }
        $run = $stepResult->getRunNumber();
        $cached = $stepResult->isCachedRun() ? 1 : 0;
        $step = $stepResult->getStepNumber();

        $testRoot = $this->testInfo->getRootDirectory();
        $testId = $this->testInfo->getId();

        $localPaths = new TestPaths($testRoot, $run, $cached, $step);
        $nameOnlyPaths = new TestPaths("", $run, $cached, $step);
        $url_friendly_dir = str_replace('./results/', '/result/', $this->testInfo->getRootDirectory());
        $remotePaths = new TestPaths($this->baseUrl . $url_friendly_dir, $run, $cached, $step);


        echo "<results>\n";
        echo ArrayToXML($stepResult->getRawResults());
        echo "</results>\n";

      // links to the relevant pages
        $urlGenerator = UrlGenerator::create($this->friendlyUrls, $this->baseUrl, $testId, $run, $cached, $step);
        echo "<pages>\n";
        echo "<details>" . htmlspecialchars($urlGenerator->resultPage("details")) . "</details>\n";
        echo "<checklist>" . htmlspecialchars($urlGenerator->resultPage("performance_optimization")) . "</checklist>\n";
        echo "<breakdown>" . htmlspecialchars($urlGenerator->resultPage("breakdown")) . "</breakdown>\n";
        echo "<domains>" . htmlspecialchars($urlGenerator->resultPage("domains")) . "</domains>\n";
        echo "<screenShot>" . htmlspecialchars($urlGenerator->resultPage("screen_shot")) . "</screenShot>\n";
        echo "<opportunities>" . htmlspecialchars($urlGenerator->resultPage("experiments")) . "</opportunities>\n";
        echo "</pages>\n";

      // URLs for the relevant images
        echo "<thumbnails>\n";
        echo "<waterfall>" . htmlspecialchars($urlGenerator->thumbnail("waterfall.png")) . "</waterfall>\n";
        echo "<checklist>" . htmlspecialchars($urlGenerator->thumbnail("optimization.png")) . "</checklist>\n";
        if ($this->fileHandler->fileExists($localPaths->screenShotFile())) {
            echo "<screenShot>" . htmlspecialchars($urlGenerator->thumbnail("screen.jpg")) . "</screenShot>\n";
        }
        echo "</thumbnails>\n";

        echo "<images>\n";
        echo "<waterfall>" . htmlspecialchars($urlGenerator->generatedImage("waterfall")) . "</waterfall>\n";
        echo "<connectionView>" . htmlspecialchars($urlGenerator->generatedImage("connection")) . "</connectionView>\n";
        echo "<checklist>" . htmlspecialchars($urlGenerator->optimizationChecklistImage()) . "</checklist>\n";
        if ($this->fileHandler->fileExists($localPaths->screenShotFile())) {
            echo "<screenShot>" . htmlspecialchars($urlGenerator->getFile($nameOnlyPaths->screenShotFile())) . "</screenShot>\n";
        }
        if ($this->fileHandler->fileExists($localPaths->screenShotPngFile())) {
            echo "<screenShotPng>" . htmlspecialchars($urlGenerator->getFile($nameOnlyPaths->screenShotPngFile())) . "</screenShotPng>\n";
            if (!$this->fileHandler->fileExists($localPaths->screenShotFile())) {
                echo "<screenShot>" . htmlspecialchars($urlGenerator->getFile($nameOnlyPaths->screenShotPngFile())) . "</screenShot>\n";
            }
        }
        echo "</images>\n";
        if ($this->fileHandler->fileExists($localPaths->renderedVideoFile())) {
            echo "<video>" . htmlspecialchars($urlGenerator->getFile($nameOnlyPaths->renderedVideoFile())) . "</video>\n";
        }

      // raw results (files accessed directly on the file system, but via URL)
        echo "<rawData>\n";
        if ($this->fileHandler->gzFileExists($localPaths->devtoolsScriptTimingFile())) {
            echo "<scriptTiming>" . htmlspecialchars($urlGenerator->getGZip($nameOnlyPaths->devtoolsScriptTimingFile())) . "</scriptTiming>\n";
        }
        if ($this->fileHandler->gzFileExists($localPaths->headersFile())) {
            echo "<headers>" . $remotePaths->headersFile() . "</headers>\n";
        }
        if ($this->fileHandler->gzFileExists($localPaths->bodiesFile())) {
            echo "<bodies>" . $remotePaths->bodiesFile() . "</bodies>\n";
        }
        if ($this->fileHandler->gzFileExists($localPaths->pageDataFile())) {
            echo "<pageData>" . $remotePaths->pageDataFile() . "</pageData>\n";
        }
        if ($this->fileHandler->gzFileExists($localPaths->requestDataFile())) {
            echo "<requestsData>" . $remotePaths->requestDataFile() . "</requestsData>\n";
        }
        if ($this->fileHandler->gzFileExists($localPaths->utilizationFile())) {
            echo "<utilization>" . $remotePaths->utilizationFile() . "</utilization>\n";
        }
        echo "</rawData>\n";

      // video frames
        $progress = $stepResult->getVisualProgress();
        if (array_key_exists('frames', $progress) && is_array($progress['frames']) && count($progress['frames'])) {
            echo "<videoFrames>\n";
            foreach ($progress['frames'] as $ms => $frame) {
                echo "<frame>\n";
                echo "<time>$ms</time>\n";
                echo "<image>" .
                htmlspecialchars($urlGenerator->getFile($frame['file'], $nameOnlyPaths->videoDir())) .
                "</image>\n";
                echo "<VisuallyComplete>{$frame['progress']}</VisuallyComplete>\n";
                echo "</frame>\n";
            }
            echo "</videoFrames>\n";
        }

        $this->printAdditionalInformation($stepResult, false);
    }

    private function printViewRootStartTag($isCachedRun)
    {
        if (!$isCachedRun) {
            echo "<firstView>\n";
        } else {
            echo "<repeatView>\n";
        }
    }

    private function printViewRootEndTag($isCachedRun)
    {
        if (!$isCachedRun) {
            echo "</firstView>\n";
        } else {
            echo "</repeatView>\n";
        }
    }

  /**
   * @param int $run The run to print the tester for
   */
    private function printTester($run)
    {
        $tester = $this->testInfo->getTester($run);
        if ($tester) {
            echo "<tester>" . xml_entities($tester) . "</tester>\n";
        }
    }

  /**
   * @param TestStepResult $testResult Result Data
   * @param bool $forMedian True if the printing is for median output, false otherwise
   */
    private function printAdditionalInformation($testResult, $forMedian)
    {
        $this->printDomainBreakdown($testResult);
        $this->printMimeTypeBreakdown($testResult);
        $this->printRequests($testResult, $forMedian);
        $this->printStatusMessages($testResult);
        $this->printConsoleLog($testResult);
    }

  /**
   * @param int $infotype The kind of info to check for (see INFO_* constants)
   * @return bool True if this type of information should be printed, false otherwise
   */
    private function shouldPrintInfo($infotype)
    {
        return in_array($infotype, $this->additionalInfo, true);
    }

  /**
   * Print information about all of the requests
   * @param TestStepResult $testResult Result Data for affected run
   * @param $forMedian True if the output is for median, false otherwise
   */
    private function printRequests($testResult, $forMedian)
    {
        if (
            !$this->shouldPrintInfo(self::INFO_REQUESTS) &&
            !($forMedian && $this->shouldPrintInfo(self::INFO_MEDIAN_REQUESTS))
        ) {
            return;
        }
        echo "<requests>\n";
        $requests = $testResult->getRequests();
        foreach ($requests as &$request) {
            echo "<request number=\"{$request['number']}\">\n";
            foreach ($request as $field => $value) {
                if (!is_array($value)) {
                    echo "<$field>" . xml_entities($value) . "</$field>\n";
                }
            }
            if (array_key_exists('headers', $request) && is_array($request['headers'])) {
                echo "<headers>\n";
                if (array_key_exists('request', $request['headers']) && is_array($request['headers']['request'])) {
                    echo "<request>\n";
                    foreach ($request['headers']['request'] as $value) {
                        echo "<header>" . xml_entities($value) . "</header>\n";
                    }
                    echo "</request>\n";
                }
                if (array_key_exists('response', $request['headers']) && is_array($request['headers']['response'])) {
                    echo "<response>\n";
                    foreach ($request['headers']['response'] as $value) {
                        echo "<header>" . xml_entities($value) . "</header>\n";
                    }
                    echo "</response>\n";
                }
                echo "</headers>\n";
            }
            echo "</request>\n";
        }
        echo "</requests>\n";
    }

  /**
   * Print a breakdown of the requests and bytes by domain
   * @param TestStepResult $testResult Result data of affected run
   */
    private function printDomainBreakdown($testResult)
    {
        if (!$this->shouldPrintInfo(self::INFO_DOMAIN_BREAKDOWN)) {
            return;
        }
        echo "<domains>\n";
        $breakdown = $testResult->getDomainBreakdown();
        foreach ($breakdown as $domain => &$values) {
            echo "<domain host=\"" . xml_entities($domain) . "\">\n";
            echo "<requests>{$values['requests']}</requests>\n";
            echo "<bytes>{$values['bytes']}</bytes>\n";
            echo "<connections>{$values['connections']}</connections>\n";
            if (isset($values['cdn_provider'])) {
                echo "<cdn_provider>{$values['cdn_provider']}</cdn_provider>\n";
            }
            echo "</domain>\n";
        }
        echo "</domains>\n";
    }

  /**
   * Print a breakdown of the requests and bytes by MIME type
   * @param TestStepResult $testResult Result data of affected run
   */
    private function printMimeTypeBreakdown($testResult)
    {
        if (!$this->shouldPrintInfo(self::INFO_MIMETYPE_BREAKDOWN)) {
            return;
        }
        echo "<breakdown>\n";
        $breakdown = $testResult->getMimeTypeBreakdown();
        foreach ($breakdown as $mime => &$values) {
            echo "<$mime>\n";
            echo "<requests>{$values['requests']}</requests>\n";
            echo "<bytes>{$values['bytes']}</bytes>\n";
            echo "</$mime>\n";
        }
        echo "</breakdown>\n";
    }

  /**
   * Print any logged browser status messages
   * @param TestStepResult $testResult Result data of affected run
   */
    private function printStatusMessages($testResult)
    {
        $messages = $testResult->getStatusMessages();
        if (!$messages) {
            return;
        }
        echo "<status>\n";
        foreach ($messages as $message) {
            echo "<entry>\n";
            echo "<time>" . xml_entities($message["time"]) . "</time>\n";
            echo "<message>" . xml_entities($message["message"]) . "</message>\n";
            echo "</entry>\n";
        }
        echo "</status>\n";
    }

  /**
   * Print the console log if requested
   * @param TestStepResult $testResult Result data of affected run
   */
    private function printConsoleLog($testResult)
    {
        if (!$this->shouldPrintInfo(self::INFO_CONSOLE)) {
            return;
        }
        $consoleLog = $testResult->getConsoleLog();
        if (isset($consoleLog) && is_array($consoleLog) && count($consoleLog)) {
            echo "<consoleLog>\n";
            foreach ($consoleLog as &$entry) {
                echo "<entry>\n";
                echo "<source>" . xml_entities($entry['source']) . "</source>\n";
                echo "<level>" . xml_entities($entry['level']) . "</level>\n";
                echo "<message>" . xml_entities($entry['text']) . "</message>\n";
                echo "<url>" . xml_entities($entry['url']) . "</url>\n";
                echo "<line>" . xml_entities($entry['line']) . "</line>\n";
                echo "</entry>\n";
            }
            echo "</consoleLog>\n";
        }
    }
}

function ArrayToXML($array)
{
    $ret = '';
    if (is_array($array)) {
        foreach ($array as $key => $val) {
            if (is_numeric($key)) {
                $key = 'value';
            }
            $key = preg_replace('/[^a-zA-Z0-9\.\-_]/', '_', $key);

          /* XXX: XML tag name should not start with digit or '-' or '.'
           * so we check if key starts with an invalid XML tagname, and if this is the
           * case, we prefix the tagname with '_'.
           */
            if (preg_match('/^[a-zA-Z_]/', $key) == 0) {
                $key = "_" . $key;
            }

            $ret .= "<$key>";
            if (is_array($val)) {
                $ret .= "\n" . ArrayToXML($val);
            } else {
                $ret .= xml_entities($val);
            }
            $ret .= "</$key>\n";
        }
    }
    return $ret;
}
